package style.Leetcode.初级算法.array.array_20221013_1_存在重复元素2;

import java.util.HashMap;
import java.util.HashSet;
import java.util.Map;

/**
 * 219. 存在重复元素 II
 *
 * 给你一个整数数组  nums 和一个整数  k ，判断数组中是否存在两个 不同的索引  i  和  j ，满足 nums[i] == nums[j] 且 abs(i - j) <= k 。如果存在，返回 true ；否则，返回 false 。
 *
 * 示例  1：
 * 输入：nums = [1,2,3,1], k = 3
 * 输出：true
 * 示例 2：
 * 输入：nums = [1,0,1,1], k = 1
 * 输出：true
 * 示例 3：
 * 输入：nums = [1,2,3,1,2,3], k = 2
 * 输出：false
 *
 */
public class Solution {

    /**
     * 思路一：
     *
     */
    public boolean containsNearbyDuplicate(int[] nums, int k) {
        Map<Integer, Integer> map = new HashMap<Integer, Integer>();
        int length = nums.length;
        for (int i = 0; i < length; i++) {
            int num = nums[i];
            if (map.containsKey(num) && i - map.get(num) <= k) {
                return true;
            }
            map.put(num, i);
        }
        return false;
    }

    /**
     * 滑动窗口
     *
     */
    public boolean containsNearbyDuplicate2(int[] nums, int k) {
        HashSet<Integer> set = new HashSet<>();
        int length = nums.length;
        for (int i = 0; i < length; i++) {
            // 控制窗口的长度只能小于 k
            if (i>k){
                set.remove(nums[i-k-1]);
            }
            if (!set.add(nums[i])) {
                return true;
            }
        }
        return false;
    }


    public static void main(String[] args) {
        Solution solution = new Solution();
        System.out.println(solution.containsNearbyDuplicate2(new int[]{1,0,1,1}, 1));
    }
}
